import torch


# 矩阵乘法运算示例可以看同目录下的图片： （名称： 矩阵乘法示例_1.png/矩阵乘法规则和讲解.png）
# 关于矩阵， 矩阵乘积， 乘积之和的详细讲解请看同目录下的图片： （名称： 矩阵_矩阵乘积_乘积之和详细讲解.png）
# todo 矩阵乘法运算示例
# 定义矩阵 A
A = torch.tensor([[1, 2, 3],
                  [4, 5, 6]], dtype=torch.float32)

# 定义矩阵 B
B = torch.tensor([[7, 8],
                  [9, 10],
                  [11, 12]], dtype=torch.float32)

# 进行矩阵乘法
C = torch.matmul(A, B)

# 输出结果
print("矩阵 A:")
print(A)
print("矩阵 B:")
print(B)
print("矩阵 A 和 B 的乘积 C:")
print(C)


# 矩阵乘法运算

data1 = torch.tensor([[1, 2], [3, 4], [5, 6]])
data2 = torch.tensor([[5, 6],[7, 8]])
# 矩阵乘法方式一
data3 = data1@data2
print('data3-->', data3)
# 矩阵乘法方式二
data4 = torch.matmul(data1, data2)
print('data4-->', data4)
